package cn.jingyuan.bee.utils.io.watch.watchers;

import cn.jingyuan.bee.utils.collection.ConcurrentHashSet;
import cn.jingyuan.bee.utils.io.watch.Watcher;
import cn.jingyuan.bee.utils.lang.Assert;
import cn.jingyuan.bee.utils.thread.ThreadUtils;

import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.WatchEvent;
import java.nio.file.WatchService;
import java.util.Set;

/**
 * 延迟观察者<br>
 * 使用此观察者通过定义一定的延迟时间，解决 {@link WatchService} 多个 modify 的问题<br>
 * 在监听目录或文件时，如果这个文件有修改操作，会多次触发 modify 方法。<br>
 * 此类通过维护一个 Set 将短时间内相同文件多次 modify 的事件合并处理触发，从而避免以上问题。<br>
 * 注意：延迟只针对 modify 事件，其它事件无效
 */
public class DelayWatcher implements Watcher {

    /** Path 集合。此集合用于去重在指定 delay 内多次触发的文件 Path */
    private final Set<Path> eventSet = new ConcurrentHashSet<>();

    /** 实际处理 */
    private final Watcher watcher;

    /** 延迟，单位毫秒 */
    private final long delay;

    /**
     * 构造
     *
     * @param watcher 实际处理触发事件的监视器 {@link Watcher}，不可以是 {@link DelayWatcher}
     * @param delay 延迟时间，单位毫秒
     */
    public DelayWatcher(Watcher watcher, long delay) {
        Assert.notNull(watcher);
        if (watcher instanceof DelayWatcher) {
            throw new IllegalArgumentException("Watcher must not be a DelayWatcher");
        }
        this.watcher = watcher;
        this.delay = delay;
    }

    @Override
    public void onModify(WatchEvent<?> event, Path currentPath) {
        if (this.delay < 1) {
            this.watcher.onModify(event, currentPath);
        } else {
            onDelayModify(event, currentPath);
        }
    }

    @Override
    public void onCreate(WatchEvent<?> event, Path currentPath) {
        watcher.onCreate(event, currentPath);
    }

    @Override
    public void onDelete(WatchEvent<?> event, Path currentPath) {
        watcher.onDelete(event, currentPath);
    }

    @Override
    public void onOverflow(WatchEvent<?> event, Path currentPath) {
        watcher.onOverflow(event, currentPath);
    }

    /**
     * 触发延迟修改
     *
     * @param event 事件
     * @param currentPath 事件发生的当前 Path 路径
     */
    private void onDelayModify(WatchEvent<?> event, Path currentPath) {
        Path eventPath = Paths.get(currentPath.toString(), event.context().toString());
        if (eventSet.contains(eventPath)) {
            // 此事件已经被触发过，后续事件忽略，等待统一处理。
            return;
        }

        // 事件第一次触发，此时标记事件，并启动处理线程延迟处理，处理结束后会删除标记
        eventSet.add(eventPath);
        startHandleModifyThread(event, currentPath);
    }

    /**
     * 开启处理线程
     *
     * @param event 事件
     * @param currentPath 事件发生的当前 Path 路径
     */
    private void startHandleModifyThread(final WatchEvent<?> event, final Path currentPath) {
        ThreadUtils.execute(new Runnable() {

            @Override
            public void run() {
                ThreadUtils.sleep(delay);
                eventSet.remove(Paths.get(currentPath.toString(), event.context().toString()));
                watcher.onModify(event, currentPath);
            }
        });
    }

}
